package com.qianfeng.iot.blackip.service.impl;


//
//                            _ooOoo_  
//                           o8888888o  
//                           88" . "88  
//                           (| -_- |)  
//                            O\ = /O  
//                        ____/`---'\____  
//                      .   ' \\| |// `.  
//                       / \\||| : |||// \  
//                     / _||||| -:- |||||- \  
//                       | | \\\ - /// | |  
//                     | \_| ''\---/'' | |  
//                      \ .-\__ `-` ___/-. /  
//                   ___`. .' /--.--\ `. . __  
//                ."" '< `.___\_<|>_/___.' >'"".  
//               | | : `- \`.;`\ _ /`;.`/ - ` : | |  
//                 \ \ `-. \_ __\ /__ _/ .-` / /  
//         ======`-.____`-.___\_____/___.-`____.-'======  
//                            `=---='  
//  
//         .............................................  
//                  佛祖镇楼            BUG辟易  
//          佛曰:  
//                  写字楼里写字间，写字间里程序员；  
//                  程序人员写程序，又拿程序换酒钱。  
//                  酒醒只在网上坐，酒醉还来网下眠；  
//                  酒醉酒醒日复日，网上网下年复年。  
//                  但愿老死电脑间，不愿鞠躬老板前；  
//                  奔驰宝马贵者趣，公交自行程序员。  
//                  别人笑我忒疯癫，我笑自己命太贱；  


import com.qianfeng.iot.blackip.events.BlackIpChangeEvent;
import com.qianfeng.iot.blackip.pojo.Dc3BlackIp;
import com.qianfeng.iot.blackip.service.BlackIpService;
import io.github.pnoker.common.bean.R;
import io.github.pnoker.common.constant.CommonConstant;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.data.r2dbc.core.R2dbcEntityTemplate;
import org.springframework.data.relational.core.query.Criteria;
import org.springframework.data.relational.core.query.Query;
import org.springframework.stereotype.Service;
import reactor.core.publisher.Mono;

import java.time.LocalDateTime;
import java.util.Set;
import java.util.UUID;
import java.util.stream.Collectors;

/**
 * Created by jackiechan on 2022/11/1 09:53
 *
 * @author jackiechan
 * @version 1.0
 * @since 1.0
 */
@Service
public class BlackIpServiceImpl implements BlackIpService {

    private ApplicationContext context;


    @Autowired
    public void setContext(ApplicationContext context) {
        this.context = context;
    }

    private R2dbcEntityTemplate r2dbcEntityTemplate;

    @Autowired
    public void setR2dbcEntityTemplate(R2dbcEntityTemplate r2dbcEntityTemplate) {
        this.r2dbcEntityTemplate = r2dbcEntityTemplate;
    }

    @Override
    public Mono<R<Object>> addIp(Mono<Dc3BlackIp> blackIpMono) {
        //我们首先要进行数据的校验
        //我们要求ip不能为空,并且必须符合ip的规范
        //如果ip符合我们的要求,就添加进去, 不复合我们的要求就返回错误数据
        //0.0.0.0-255.255.255.255
        //其实还有个事情要注意,我们没有判断这个ip在数据库中是不是存在
        return  blackIpMono.filter(dc3BlackIp -> dc3BlackIp.getIp() != null && dc3BlackIp.getIp().matches(CommonConstant.Validate.IP_REG))
                //在插入数据之前要先设置主键,主键是一个非自增的字符串
                .doOnNext(dc3BlackIp -> {
                    dc3BlackIp.setId(UUID.randomUUID().toString().replace("-", ""));
                    dc3BlackIp.setCreateTime(LocalDateTime.now());//设置创建时间
                }).flatMap(dc3BlackIp -> {//当前map中的数据已经带有id了
                    return r2dbcEntityTemplate.insert(dc3BlackIp).map(dc3BlackIp1 -> {
                        //现在代表ip添加完成,我们应该发送一个事件通知
                        context.publishEvent(new BlackIpChangeEvent());
                        return R.ok();});
                }).defaultIfEmpty(R.fail("ip不符合要求"));


    }

    @Override
    public Mono<R<Set<String>>> findAllEnableIp() {
//        Flux<Dc3BlackIp> dc3BlackIpFlux1 = r2dbcEntityTemplate.select(Dc3BlackIp.class)
//                .from("dc3_black_ip")
//                .matching(Query.query(Criteria.where("enable").is(1).and("deleted").is(0))).all();

        return r2dbcEntityTemplate.select(Query.query(Criteria.where("enable").is(1).and("deleted").is(0)), Dc3BlackIp.class)//查询所有的数据
                .map(Dc3BlackIp::getIp)//将返回结果中只保留ip
                .collect(Collectors.toSet())//将所有的ip放到一个集合中
                .map(list->R.ok(list));//将这个集合放到R里面

    }
}
